package ui;

import java.awt.Color;
import java.awt.Graphics;
import java.awt.Insets;
import java.awt.event.KeyAdapter;
import java.awt.event.KeyEvent;
import java.io.File;
import java.io.FileNotFoundException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;

import javax.swing.JTextPane;
import javax.swing.text.Element;
import javax.swing.text.SimpleAttributeSet;
import javax.swing.text.StyleConstants;
import javax.swing.text.StyledDocument;

import util.FileUtil;

//编辑界面
public class EditPane extends JTextPane {
	//样式文档对象hvhgy
	protected StyledDocument doc;
	//构造一个文本格式器
	protected SyntaxFormatter formatter = new SyntaxFormatter("cmm.stx");
	private SimpleAttributeSet quotAttr = new SimpleAttributeSet();
	//保存文档改变的开始位置
	private int docChangeStart = 0;
	//保存文档改变的长度
	private int docChangeLength = 0;
	//定义文档中行数文本的外观属性
	private SimpleAttributeSet lineAttr = new SimpleAttributeSet();
	
	public EditPane(File file){
		this.setText(FileUtil.readFile(file));
		this.doc = getStyledDocument();
		//设置文档页边距
		this.setMargin(new Insets(3, 40, 0, 0));
		syntaxParse();
		//按键监听器，松开时进行语法分析
		this.addKeyListener(new KeyAdapter() {
			public void keyReleased(KeyEvent e){
				syntaxParse();
			}
		});
	}
	
	public void syntaxParse(){
		try{
			//获取文档根元素，即全部内容
			Element root = doc.getDefaultRootElement();
			//获取光标位置
			int cursorPos = this.getCaretPosition();
			int line = root.getElementIndex(cursorPos);
			//获取光标所在行
			Element para = root.getElement(line);
			//定义光标所在行的行头在文档中位置
			int start = para.getStartOffset();
			//如果文档修改位置比当前行还前
			if (start > docChangeStart){
				start = docChangeStart;
			}
			//定义被修改部分的长度
			int length = para.getEndOffset() - start;
			if (length < docChangeLength) {
				length = docChangeLength + 1;
			}
			//取出被修改的字符串
			String s = doc.getText(start, length);
			//空格 点号作为分隔符
			String[] tokens = s.split("\\s+|\\.|\\(|\\)|\\{|\\}|\\[|\\]");
			//定义当前分析单词在s中的开始位置
			int curStart = 0;
			boolean isQuot = false;
			for(String token : tokens){
				//找出当前分析单词在s字符串的中位置
				int tokenPos = s.indexOf(token , curStart);
				if (isQuot && (token.endsWith("\"") || token.endsWith("\'"))) {
					doc.setCharacterAttributes(start + tokenPos, token.length(), quotAttr, false);
					isQuot = false;
				}
				else if (isQuot && !(token.endsWith("\"") || token.endsWith("\'"))) {
					doc.setCharacterAttributes(start + tokenPos, token.length(), quotAttr, false);
				}
				else if ((token.startsWith("\"") || token.startsWith("\'"))
						&& (token.endsWith("\"") || token.endsWith("\'"))) {
					doc.setCharacterAttributes(start + tokenPos, token.length(), quotAttr, false);
				}
				else if ((token.startsWith("\"") || token.startsWith("\'"))
						&& !(token.endsWith("\"") || token.endsWith("\'"))) {
					doc.setCharacterAttributes(start + tokenPos, token.length(), quotAttr, false);
					isQuot = true;
				}
				else {
					//使用格式器对当前单词设置颜色
					formatter.setHighLight(doc , token , start + tokenPos, token.length());
				}
				//开始分析下一个单词
				curStart = tokenPos + token.length();
			}
		}catch(Exception e){
			e.printStackTrace();
		}
	}
	
	//
	//重画该组件，设置行号
	public void paint(Graphics g){
		super.paint(g);
		Element root = doc.getDefaultRootElement(); 
		//获得行号
		int line = root.getElementIndex(doc.getLength());
		//设置颜色
		g.setColor(new Color(230, 230, 230));
		//绘制行数矩形框
		g.fillRect(0, 0, this.getMargin().left - 10, getSize().height);
		//设置行号的颜色
		g.setColor(new Color(40, 40, 40));
		//每行绘制一个行号
		for (int count = 0, j = 1; count <= line; count++, j++) {
			g.drawString(String.valueOf(j), 3, 
					(int)((count + 1) * 1.5020 * StyleConstants.getFontSize(lineAttr)));
		}
	}
}

class SyntaxFormatter{
	//一个Map保存关键字和颜色的对应关系
	private Map<SimpleAttributeSet,ArrayList> attMap = new HashMap<SimpleAttributeSet,ArrayList>();
	//定义正常文本的外观属性
	SimpleAttributeSet normalAttr = new SimpleAttributeSet();
	public SyntaxFormatter(String syntaxFile){
		//设置正常文本的颜色大小
		StyleConstants.setForeground(normalAttr, Color.black);
		//创建一个Scanner对象，根据语法文件加载颜色信息
		Scanner scaner = null;
		try{
			scaner = new Scanner(new File(syntaxFile));
		}catch(FileNotFoundException e){
			throw new RuntimeException("丢失语法文件： "+e.getMessage());
		}
		int color = -1;
		ArrayList<String> keywords = new ArrayList<String>();
		//读取语法文件的内容行
		while(scaner.hasNextLine()){
			String line = scaner.nextLine();
			//如果当前行以#开头
			if (line.startsWith("#")) {
				if (keywords.size() > 0 && color > -1) {
					//取出当前行的颜色值，并封装成SimpleAttributeSet对象
					SimpleAttributeSet att = new SimpleAttributeSet();
					StyleConstants.setForeground(att, new Color(color));
					//将当前颜色和关键字List对应起来
					attMap.put(att , keywords);
				}
				//重新创建新的关键字List，为下一个语法格式准备
				keywords = new ArrayList<String>();
				color = Integer.parseInt(line.substring(1) , 16);
			} 
			else {
				//对于普通行，每行内容添加到关键字List里
				if (line.trim().length() > 0) {
					keywords.add(line.trim());
				}
			}
		}
		scaner.close();
		//把最后的关键字和颜色对应起来
		if (keywords.size() > 0 && color > -1) {
			SimpleAttributeSet att = new SimpleAttributeSet();
			StyleConstants.setForeground(att, new Color(color));
			attMap.put(att , keywords);
		}
	}
	
	//返回该格式器里正常文本的外观属性
	public SimpleAttributeSet getNormalAttributeSet(){
		return normalAttr;
	}
	
	//设置语法高亮
	public void setHighLight(StyledDocument doc, String token, int start, int length){
		//保存需要对当前单词对应的外观属性
		SimpleAttributeSet currentAttributeSet = null;
		outer :
			for (SimpleAttributeSet att : attMap.keySet()){
				//取出当前颜色对应的所有关键字
				ArrayList keywords = attMap.get(att);
				//遍历所有关键字
				for (Object keyword : keywords){
					//如果该关键字与当前单词相同
					if (keyword.toString().equals(token)){
						//跳出循环，并设置当前单词对应的外观属性
						currentAttributeSet = att;
						break outer;
					}
				}
			}
		//如果当前单词对应的外观属性不为空
		if (currentAttributeSet != null){
			//设置当前单词的颜色
			doc.setCharacterAttributes(start, length, currentAttributeSet, false);
		} 
		else {//否则使用普通外观来设置该单词
			doc.setCharacterAttributes(start, length, normalAttr, false);
		}
	}
}

